java/Graph Editor/src/controllers/actions/DeleteNode.java

   1 package controllers.actions;
   2
   3 import javax.swing.undo.AbstractUndoableEdit;
   4
   5 import models.GraphEdge;
   6 import models.GraphVertex;
   7 import controllers.*;
   8
   9 public class DeleteNode extends AbstractUndoableEdit {
  10         private static final long serialVersionUID = -7929100929760152271L;
  11         private GraphVertex o;
  12         private SelectionController selectionController;
  13
  14         private GraphEdge[] affectedEdges;
  15
  16         public DeleteNode(SelectionController selectionController)
  17         {
  18                 this.selectionController = selectionController;
  19                 this.o = selectionController.getSelectedVertex();
  20                 redo();
  21         }
  22
  23         private void removeAffectedEdges()
  24         {
  25                 selectionController.getModel().removeEdge(affectedEdges);
  26         }
  27
  28         private void addAffectedEdges()
  29         {
  30                 selectionController.getModel().addEdge(affectedEdges);
  31         }
  32
  33         private void getAffectedEdges(){
  34                 GraphEdge[] arr = selectionController.getModel().getEdges();
  35                 int count = 0;
  36                 for(int i=0; i < arr.length; i++)
  37                         if(arr[i].EdgeOf(o))
  38                                 count++;
  39                 count = 0;
  40                 affectedEdges = new GraphEdge[count];
  41                 for(int i=0; i < arr.length; i++)
  42                         if(arr[i].EdgeOf(o))
  43                                 affectedEdges[count++] = arr[i];
  44         }
  45
  46
  47         public void redo()
  48         {
  49                 getAffectedEdges();
  50                 removeAffectedEdges();
  51                 selectionController.removeSelected();
  52                 selectionController.getModel().removeGraph(o);
  53         }
  54
  55         public void undo()
  56         {
  57                 addAffectedEdges();
  58                 selectionController.getModel().addGraph(o);
  59                 selectionController.setSelected(o);
  60         }
  61 }